class Solution {
    int dx[4] = {0, 0 , -1, 1}; // i点加dx，dy就是i点的上下左右下标
    int dy[4] = {1, -1 , 0, 0};
    
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color) {
        int prev = image[sr][sc];
        if(prev == color)
            return image;
 
        int m = image.size(), n = image[0].size(); // 获取大小防止越界
        queue<pair<int, int>> q; // 存下标
        q.push({sr, sc});
        while(!q.empty()) // 模拟层序遍历
        {
            auto [a, b] = q.front(); // 拿到队头的元素（一个pair）
            q.pop();
            image[a][b] = color; // 拿到的元素染上新颜色
            for(int i = 0; i < 4; ++i)
            {
                int x = a + dx[i], y = b + dy[i];
                if(x >= 0 && x < m && y >= 0 && y < n && image[x][y] == prev)
                {
                    q.push({x, y});
                }
            }
        }
        return image;
    }
};